package com.linyaonan.leetcode.easy._617;

/**
 *
 * 给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。
 *
 * 你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。
 *
 * 示例 1:
 *
 * 输入:
 * 	Tree 1                     Tree 2
 *           1                         2
 *          / \                       / \
 *         3   2                     1   3
 *        /                           \   \
 *       5                             4   7
 * 输出:
 * 合并后的树:
 * 	     3
 * 	    / \
 * 	   4   5
 * 	  / \   \
 * 	 5   4   7
 * 注意: 合并必须从两个树的根节点开始。
 *
 *
 * @author: Lin
 * @date: 2020/1/4
 */
public class MergeTwoTrees {
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        if (t1 == null) {
            return t2;
        }
        if (t2 == null) {
            return t1;
        }
        merge(t1, t2, t1, true);
        return t1;
    }

    private void merge(TreeNode node1, TreeNode node2, TreeNode p, boolean isLeft) {
        if (node1 == null && node2 == null) {
            return;
        } else if (node1 == null) {
            if (isLeft) {
                p.left = node2;
            } else {
                p.right = node2;
            }
        } else if (node2 != null) {
            node1.val += node2.val;
            merge(node1.left, node2.left, node1, true);
            merge(node1.right, node2.right, node1, false);
        }
    }
}
